/**
 * 二进制写包优点：
 *    1.提高传输效率（节省空间)
 *	  2.简化数据解析过程（固定大小跟顺序)
 *    3.确保跨平台统一性
 *    4.适应网络协议要求
 */
package base

import (
	"bytes"
	"encoding/binary"
	"errors"
	"zinx/interfa"
	"zinx/utils"
)

var defaultHeaderLen uint32 = 0

// DataPack 封包 拆包实例
type DataPack struct{}

func NewDataPack() interfa.IDataPack {
	return &DataPack{}
}

func (dp *DataPack) GetHeadLen() uint32 {
	return defaultHeaderLen
}

func (dp *DataPack) Pack(msg interfa.IMessage) ([]byte, error) {
	dataBuff := bytes.NewBuffer([]byte{})

	if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetDataLen()); err != nil {
		return nil, err
	}

	if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetMsgID()); err != nil {
		return nil, err
	}

	if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetData()); err != nil {
		return nil, err
	}

	return dataBuff.Bytes(), nil
}

func (dp *DataPack) UnPack(binaryData []byte) (interfa.IMessage, error) {
	//创建一个从输入二进制数据的ioReader
	dataBuff := bytes.NewReader(binaryData)

	//只解压head的信息，得到dataLen和msgID
	msg := &Message{}

	//读dataLen
	if err := binary.Read(dataBuff, binary.LittleEndian, &msg.DataLen); err != nil {
		return nil, err
	}

	//读msgID
	if err := binary.Read(dataBuff, binary.LittleEndian, &msg.ID); err != nil {
		return nil, err
	}

	//判断dataLen的长度是否超出我们允许的最大包长度
	if utils.GlobalObject.MaxPacketSize > 0 && msg.DataLen > utils.GlobalObject.MaxPacketSize {
		return nil, errors.New("too large msg data received")
	}

	//这里只需要把head的数据拆包出来就可以了，然后再通过head的长度，再从conn读取一次数据
	return msg, nil
}
